Skip to content

add thinking content token#2

Open
top2group wants to merge 1 commit into
mainfrom
yu_fix
Open

add thinking content token#2
top2group wants to merge 1 commit into
mainfrom
yu_fix

Conversation

@top2group
Copy link
Copy Markdown

textbook_writer/problem_curator/main.py中,在run_prompt函数中async for event in runner.run_async内:
event.usage_metadata.prompt_token_count是传给大模型的prompt的token数
event.usage_metadata.candidates_token_count是收到大模型回复的正式内容的token数
event.usage_metadata.total_token_count对应总token数,等于prompt_token_count+candidates_token_count
本次改动增加了思维链的token数的输出,可通过event.usage_metadata.thoughts_token_count来获取

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant